package leetcode.dynamic;

import java.util.Arrays;

/**
 * 给你两个单词 word1 和 word2， 请返回将 word1 转换成 word2 所使用的最少操作数  。
 * <p>
 * 你可以对一个单词进行如下三种操作：
 * <p>
 * 插入一个字符
 * 删除一个字符
 * 替换一个字符
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：word1 = "horse", word2 = "ros"
 * 输出：3
 * 解释：
 * horse -> rorse (将 'h' 替换为 'r')
 * rorse -> rose (删除 'r')
 * rose -> ros (删除 'e')
 * 示例 2：
 * <p>
 * 输入：word1 = "intention", word2 = "execution"
 * 输出：5
 * 解释：
 * intention -> inention (删除 't')
 * inention -> enention (将 'i' 替换为 'e')
 * enention -> exention (将 'n' 替换为 'x')
 * exention -> exection (将 'n' 替换为 'c')
 * exection -> execution (插入 'u')
 * <p>
 * <p>
 * 提示：
 * <p>
 * 0 <= word1.length, word2.length <= 500
 * word1 和 word2 由小写英文字母组成
 */
public class LeetCode72_MinDistance2 {
    public int minDistance(String word1, String word2) {
        return f(word1.toCharArray(), word2.toCharArray(), word1.length() - 1, word2.length() - 1, 1, 1, 1);
    }

    public int f(char[] s1, char[] s2, int i, int j, int ins, int del, int change) {
        if (i < 0) {
            return (j + 1) * ins;
        }
        if (j < 0) {
            return del;
        }
        if (s1[i] != s2[j]) {
            int ans1 = del + f(s1, s2, i - 1, j, ins, del, change);
            int ans2 = del + f(s1, s2, i - 1, j - 1, ins, del, change) + ins;
            int ans3 = change + f(s1, s2, i - 1, j - 1, ins, del, change);
            int ans4 = ins + f(s1, s2, i, j - 1, ins, del, change);

            return Math.min(Math.min(ans1, ans2), Math.min(ans3, ans4));
        } else {
            return f(s1, s2, i - 1, j - 1, ins, del, change);
        }
    }

    public int minDistance2(String word1, String word2) {
        int[][] dp = new int[word1.length()][word2.length()];
        for (int[] t : dp) {
            Arrays.fill(t, -1);
        }
        return f2(word1.toCharArray(), word2.toCharArray(), word1.length() - 1, word2.length() - 1, 1, 1, 1, dp);
    }

    public int f2(char[] s1, char[] s2, int i, int j, int ins, int del, int change, int[][] dp) {
        if (i < 0) {
            return (j + 1) * ins;
        }
        if (j < 0) {
            return del * (i + 1);
        }

        if (dp[i][j] != -1) {
            return dp[i][j];
        }

        int ans;
        if (s1[i] != s2[j]) {
            int ans1 = del + f2(s1, s2, i - 1, j, ins, del, change, dp);
            int ans2 = del + f2(s1, s2, i - 1, j - 1, ins, del, change, dp) + ins;
            int ans3 = change + f2(s1, s2, i - 1, j - 1, ins, del, change, dp);
            int ans4 = ins + f2(s1, s2, i, j - 1, ins, del, change, dp);

            ans = Math.min(Math.min(ans1, ans2), Math.min(ans3, ans4));
        } else {
            ans = f2(s1, s2, i - 1, j - 1, ins, del, change, dp);
        }
        dp[i][j] = ans;

        return ans;
    }
}
